def load_knowledge():
    # 将每个文件的内容放在一个列表中
    documents = [
        '同事信息：',
        '金晶 女 50岁',
        '林超 男 30岁',
        '明科 男 20岁',
        '兴趣爱好：',
        '金晶 喜欢购物 刷剧',
        '林超 喜欢钓鱼',
        '明科 喜欢编程'
    ]
    return documents


def search(query, documents):
    results = [doc for doc in documents if query in doc]
    return results


def main():
    documents = load_knowledge()
    print("知识库加载完成，输入'quit'退出")
    while True:
        user_input = input("请输入您的问题: ")
        if user_input.lower() == 'quit':
            print("程序结束")
            break

        results = search(user_input, documents)
        if results:
            for result in results:
                print(f"相关文档: {result}")
        else:
            print("没有找到相关文档")


if __name__ == '__main__':
    main()
